#该问题发生于当我使用yolo官方模型推理时，其生成的txt文件有我不需要的类和类发生错位
#所以需要更改生成txt文件的内容，但此时我还需要知道需要的类的原始类中的下标
#由于labelImg的label文件存放位置还需有个class.txt文件，该脚本也可以生成
if __name__ == "__main__":
    content = [ 'person', 'bicycle', 'car', 'motorcycle', 'airplane', 'bus', 'train', 'truck', 'boat', 'traffic light',
         'fire hydrant', 'stop sign', 'parking meter', 'bench', 'bird', 'cat', 'dog', 'horse', 'sheep', 'cow',
         'elephant', 'bear', 'zebra', 'giraffe', 'backpack', 'umbrella', 'handbag', 'tie', 'suitcase', 'frisbee',
         'skis', 'snowboard', 'sports ball', 'kite', 'baseball bat', 'baseball glove', 'skateboard', 'surfboard',
         'tennis racket', 'bottle', 'wine glass', 'cup', 'fork', 'knife', 'spoon', 'bowl', 'banana', 'apple',
         'sandwich', 'orange', 'broccoli', 'carrot', 'hot dog', 'pizza', 'donut', 'cake', 'chair', 'couch',
         'potted plant', 'bed', 'dining table', 'toilet', 'tv', 'laptop', 'mouse', 'remote', 'keyboard', 'cell phone',
         'microwave', 'oven', 'toaster', 'sink', 'refrigerator', 'book', 'clock', 'vase', 'scissors', 'teddy bear',
         'hair drier', 'toothbrush' ]
    
    index = content.index("dog")
    print(index)

    # ans = ""
    # for i in content:
    #     ans = ans + i + "\n"
    # print(ans)
    # with open('./a.txt', 'w') as f:
    #     f.write(ans)
